package com.xie.leetcode.array;

//6. Z 字形变换
//        将一个给定字符串 s 根据给定的行数 numRows ，以从上往下、从左到右进行 Z 字形排列。
//
//        比如输入字符串为 "PAYPALISHIRING" 行数为 3 时，排列如下：
//
//        P   A   H   N
//        A P L S I I G
//        Y   I   R
//        之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："PAHNAPLSIIGYIR"。
//
//        请你实现这个将字符串进行指定行数变换的函数：
//
//        string convert(string s, int numRows);
//
//
//        示例 1：
//
//        输入：s = "PAYPALISHIRING", numRows = 3
//        输出："PAHNAPLSIIGYIR"
//        示例 2：
//        输入：s = "PAYPALISHIRING", numRows = 4
//        输出："PINALSIGYAHRPI"
//        解释：
//        P     I    N
//        A   L S  I G
//        Y A   H R
//        P     I
//        示例 3：
//
//        输入：s = "A", numRows = 1
//        输出："A"
//
//
//        提示：
//
//        1 <= s.length <= 1000
//        s 由英文字母（小写和大写）、',' 和 '.' 组成
//        1 <= numRows <= 1000

/**
 * @author xiezhendong
 * @date 2021/10/28
 */
public class ConvertZ {

    public static void main(String[] args) {
        ConvertZ convertZ = new ConvertZ();
        System.out.println(convertZ.convert("PAYPALISHIRING", 3));
        System.out.println(convertZ.convert("PAYPALISHIRING", 4));
        System.out.println(convertZ.convert("A", 1));
        System.out.println(convertZ.convert("ABC", 2));
        System.out.println(convertZ.convert("ABC", 3));
    }

    public String convert(String s, int numRows) {
        int numZ = numRows - 2;
        if (numZ < 0) {
            return s;
        }
        StringBuilder str = new StringBuilder();

        int len = s.length();
        int num = 0;

        while (len > 0) {
            len = len - numRows;
            num++;
            if (len <= 0) {
                break;
            }
            len = len - numZ;
            num = num + numZ;
            if (len <= 0) {
                break;
            }
        }

        String[][] vals = new String[numRows][num];
        char[] chars = s.toCharArray();
        int cel = 0;
        int row = 0;
        for (int i = 0; i < chars.length; i++) {
            vals[cel][row] = String.valueOf(chars[i]);
            cel++;
            if (cel % numRows == 0) {
                cel = 0;
                row++;
                for (int i1 = 0; i1 < numZ; i1++) {
                    if (chars.length - 1 < i + 1) {
                        break;
                    }
                    i++;
                    vals[numRows - 2 - i1][row] = String.valueOf(chars[i]);
                    row++;
                }
            }
        }
        for (String[] val : vals) {
            for (String s1 : val) {
                if (s1 != null) {
                    str.append(s1);
                }
            }
        }
        return str.toString();
    }
}
